package com.skh.string;

import java.util.*;

/**
 * @Author: skh
 * @Date: 2020/3/31 14:24
 * @Description: 变位词组
 */
public class GroupAnagrams {

    /**
     * 编写一种方法，对字符串数组进行排序，将所有变位词组合在一起。变位词是指字母相同，但排列不同的字符串。
     *
     * 注意：本题相对原题稍作修改
     *
     * 示例:
     *
     * 输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
     * 输出:
     * [
     *   ["ate","eat","tea"],
     *   ["nat","tan"],
     *   ["bat"]
     * ]
     * 说明：
     *
     * 所有输入均为小写字母。
     * 不考虑答案输出的顺序。
     *
     */

    /**
     * 对strs中的每一个字符串strs[i]进行排序（桶排序），然后判断Map中是否有该关键字（例如bat，排序后为abt，判断Map中是否有abt）
     * 1.如果没有该关键字，则新建一个ArrayList，将strs[i]加入ArrayList，然后将ArrayList加入结果res。并将排序后的字符串作为key，其所属ArrayList存储在res中的位置作为value。
     * 2.如果有该关键字，则取出所属ArrayList在res中的位置，并将所属ArrayList从res中取出，将strs[i]加入ArrayList。
     */
    public List<List<String>> groupAnagrams(String[] strs) {
        List<List<String>> result = new ArrayList<>();
        Map<String, Integer> map = new HashMap<>();

        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String s = String.valueOf(chars);
            if (!map.containsKey(s)) {
                List<String> temp = new ArrayList<>();
                temp.add(str);
                map.put(s, result.size());
                result.add(temp);
            } else {
                Integer index = map.get(s);
                result.get(index).add(str);
            }
        }
        return result;
    }

    //使用java8语法来完成
    public List<List<String>> groupAnagrams1(String[] strs) {
        List<List<String>> result = new ArrayList<>();
        Map<String, List<String>> map = new HashMap<>();

        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String s = String.valueOf(chars);
            //如果key不存在,则执行function函数,即新建数组,否则直接get(key)然后add(str).
            map.computeIfAbsent(s,x->new ArrayList()).add(str);
        }

        //遍历map
        for (Map.Entry<String, List<String>> entry : map.entrySet()) {
            result.add( entry.getValue());
        }

        return result;
    }
}
